package com.example.algorithm;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.Comparator;
import java.util.List;

/**
 * 查询主机负载最大编号
 *
 * @DATE 2024/6/14
 **/
public class GetHighestLoadHost {

    public static void main(String[] args) {

    }

    public int[] findHighestHost(int serverNum, Message[] messages) {
        Arrays.sort(messages, Comparator.comparingInt(m -> m.time));
        // times[i]表示第i台主机，下次可处理请求的时刻
        int[] times = new int[serverNum];
        // 初始值设置为1
        Arrays.fill(times, 1);
        // load[i]表示第i台主机的负载值
        int[] loads = new int[serverNum];
        // 轮询主机索引，从第1台主机开始
        int start = 0;
        for (Message message : messages) {
            boolean flag = false;
            int j = 0;
            for (int i = 0; i < serverNum; i++) {
                j = (start + i) % serverNum;
                if (times[j] <= message.time) {
                    // 当前主机j下次可以处理请求的时刻值 <= 当前消息时刻值，满足处理条件
                    times[j] = message.time + message.load;
                    loads[j] += message.load;
                    flag = true;
                    // 找到满足条件的主机编号j后，直接跳出当前for循环，轮询寻找下次消息处理的主机编号
                    break;
                }
            }

            // 轮询寻找下次主机编号
            if (flag) {
                start = (j + 1) % serverNum;
            }
        }

        // 找出最大负载结果
        List<Integer> ans = new ArrayList<>();
        ans.add(0);
        for (int i = 0; i < serverNum; i++) {
            if (loads[i] > loads[ans.get(0)]) {
                ans.clear();
                ans.add(i);
            } else if (loads[i] == loads[ans.get(0)]) {
                ans.add(i);
            }
        }

        // 由于主机编号从1开始，而ans中值从0开始，所以这里需要自增1
        return ans.stream().mapToInt(i -> i + 1).toArray();
    }

    static class Message {
        int time;
        int load;
    }
}
